package common;

import com.mysql.jdbc.jdbc2.optional.MysqlDataSource;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * 数据库jdbc操作工具类，主要提供
 * 1.getConnection：获取数据库连接对象
 * 2.close：释放数据库资源
 */
public class DBUtil {
    // 数据库连接url
    private static final String URL = "jdbc:mysql://127.0.0.1:3306/oj_database?characterEncoding=utf8&useSSL=false";
    // 数据库连接账号
    private static final String USERNAME = "root";
    // 数据库连接密码
    private static final String PASSWORD = "123456";

    // 数据库连接池
    private static volatile DataSource dataSource = null;

    /**
     * 获取数据库连接池：设计为多线程安全的单例模式，双重校验锁的写法
     * @return
     */
    private static DataSource getDataSource() {
        // 如果连接池还没有实例化
        if (dataSource == null) {
            // 可能发生多线程操作共享变量，需要保证线程安全，加锁
            synchronized (DBUtil.class) {
                // 多个线程申请锁，只实例化一次，满足单例模式
                if (dataSource == null) {
                    // 创建MySQL数据库连接池
                    MysqlDataSource mysqlDataSource = new MysqlDataSource();
                    // 设置url
                    mysqlDataSource.setURL(URL);
                    // 设置账号
                    mysqlDataSource.setUser(USERNAME);
                    // 设置密码
                    mysqlDataSource.setPassword(PASSWORD);
                    // 赋值给静态变量，这样只有一份
                    dataSource = mysqlDataSource;
                }
            }
        }
        return dataSource;
    }

    /**
     * 获取数据库连接对象
     * @return
     * @throws SQLException
     */
    public static Connection getConnection() throws SQLException {
        // 从同一个数据库连接池中获取数据库连接对象，可以达到复用连接的目的
        return getDataSource().getConnection();
    }

    /**
     * 释放数据库资源
     * @param connection    数据库连接资源
     * @param statement     操作命令资源
     * @param resultSet     结果集资源
     */
    public static void close(Connection connection, PreparedStatement statement, ResultSet resultSet) {
        // 如果结果集对象不为空
        if (resultSet != null) {
            try {
                // 关闭结果集对象
                resultSet.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        // 如果操作命令对象不为空
        if (statement != null) {
            try {
                // 关闭操作命令对象
                statement.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        // 如果连接对象不为空
        if (connection != null) {
            try {
                // 关闭连接对象
                connection.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
}
